package com.jieao.common.utils.file;

import com.aliyun.oss.OSSClient;
import com.aliyun.oss.model.ObjectMetadata;
import com.aliyun.oss.model.PutObjectResult;
import com.jieao.common.exception.base.BaseException;
import com.jieao.common.exception.file.FileException;
import com.jieao.common.exception.file.FileTypeException;
import com.jieao.common.utils.StringUtils;
import com.jieao.common.utils.file.constant.FileTypeEnum;
import com.jieao.common.utils.file.constant.OSSProperty;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.Calendar;
import java.util.Date;
import java.util.Random;

/**
 * @Description
 * @Author weifanl
 * @Date2021/3/25 17:04
 **/
@Configuration
public class OSSFileUtils {
    /**
     * 日志
     */
    private static Logger logger = LoggerFactory.getLogger(OSSFileUtils.class);

    /**
     * 获取阿里云OSS客户端对象
     *
     * @return ossClient
     */
    public static OSSClient getOSSClient() {
        return new OSSClient(OSSProperty.endpoint, OSSProperty.accessKeyId, OSSProperty.accessKeySecret);
    }


    private static String getKey(String url) {
        return url.replaceFirst("https://", "")
                .replaceFirst("http://", "")
                .replaceFirst(OSSProperty.imgBucketName + "." + OSSProperty.endpoint, "")
                .replaceFirst(OSSProperty.videoBucketName + "." + OSSProperty.endpoint, "")
                .replaceFirst(OSSProperty.fileBucketName + "." + OSSProperty.endpoint, "");
    }

    /**
     * 上传至OSS
     *
     * @param file   上传文件（文件全路径如：D:\\image\\cake.jpg）
     * @param folder 模拟文件夹名 如"qj_nanjing/"
     * @return String 返回的唯一MD5数字签名
     */
    public static String uploadFile(OSSClient ossClient, File file, String folder) {
        String resultStr = null;
        try {
            // 以输入流的形式上传文件
            InputStream is = new FileInputStream(file);
            // 文件名
            String fileName = file.getName();
            // 文件大小
            Long fileSize = file.length();
            // 文件的后缀名
            String fileExtension = fileName.substring(fileName.lastIndexOf("."));

            FileTypeEnum fileType = FileTypeEnum.getFileTypeByExtension(fileExtension);
            if (null == fileType) {
                throw new FileTypeException();
            }
            // 创建上传Object的Metadata
            ObjectMetadata metadata = new ObjectMetadata();
            // 上传的文件的长度
            metadata.setContentLength(is.available());
            // 指定该Object被下载时的网页的缓存行为
            metadata.setCacheControl("no-cache");
            // 指定该Object下设置Header
            metadata.setHeader("Pragma", "no-cache");
            // 指定该Object被下载时的内容编码格式
            metadata.setContentEncoding("utf-8");
            // 文件的MIME，定义文件的类型及网页编码，决定浏览器将以什么形式、什么编码读取文件。如果用户没有指定则根据Key或文件名的扩展名生成，
            metadata.setContentType(fileType.getContentType());
            // 指定该Object被下载时的名称（指示MINME用户代理如何显示附加的文件，打开或下载，及文件名称）
            metadata.setContentDisposition("filename/filesize=" + fileName + "/" + fileSize + "Byte.");
            //处理文件夹
            folder = dealFolder(folder);
            // 上传文件 (上传文件流的形式)
            PutObjectResult putResult = ossClient.putObject(fileType.getBucketName(), folder + fileName, is, metadata);
            // 解析结果
            resultStr = putResult.getETag();
        } catch (Exception e) {
            e.printStackTrace();
            logger.error("上传阿里云OSS服务器异常." + e.getMessage(), e);
        }
        return resultStr;
    }

    /**
     * 上传至OSS
     *
     * @param ossClient
     * @param file      文件流
     * @param folder    模拟文件夹 目前以前端路由地址作为模拟文件夹
     * @return String 返回的唯一MD5数字签名
     */
    public static String uploadMultipartFile(OSSClient ossClient, MultipartFile file, String folder) {
        String resultStr = null;
        try {
            // 以输入流的形式上传文件
            InputStream is = file.getInputStream();
            String name = file.getOriginalFilename();

            // 文件的后缀名
            String fileExtension = "";
            if (name.indexOf(".") == -1) {
                fileExtension = "." + MimeTypeUtils.getExtension(file.getContentType());
            } else {
                fileExtension = name.substring(name.lastIndexOf("."));
            }
            // 文件名
            String fileName = generateRandomFilename() + fileExtension;
            // 文件大小
            Long fileSize = file.getSize();

            FileTypeEnum fileType = FileTypeEnum.getFileTypeByExtension(fileExtension);
            if (null == fileType) {
                throw new FileTypeException();
            }
            // 创建上传Object的Metadata
            ObjectMetadata metadata = new ObjectMetadata();
            // 上传的文件的长度
            metadata.setContentLength(is.available());
            // 指定该Object被下载时的网页的缓存行为
            metadata.setCacheControl("no-cache");
            // 指定该Object下设置Header
            metadata.setHeader("Pragma", "no-cache");
            // 指定该Object被下载时的内容编码格式
            metadata.setContentEncoding("utf-8");
            // 文件的MIME，定义文件的类型及网页编码，决定浏览器将以什么形式、什么编码读取文件。如果用户没有指定则根据Key或文件名的扩展名生成，
            // 如果没有扩展名则填默认值application/octet-stream
            metadata.setContentType(fileType.getContentType());
            // 指定该Object被下载时的名称（指示MINME用户代理如何显示附加的文件，打开或下载，及文件名称）
            metadata.setContentDisposition("filename/filesize=" + fileName + "/" + fileSize + "Byte.");
            //处理文件夹
            folder = dealFolder(folder);
            // 上传文件 (上传文件流的形式)
            PutObjectResult putResult = ossClient.putObject(fileType.getBucketName(), folder + fileName, is, metadata);
            // 解析结果
            Date expiration = new Date(System.currentTimeMillis() + 360 * 1000 * 24 * 365 * 10);
            String url = ossClient.generatePresignedUrl(fileType.getBucketName(), folder + fileName, expiration).toString();
            resultStr = dealResultUrl(url, fileType.getBucketName(), fileType.getCdnUrl());
        } catch (Exception e) {
            e.printStackTrace();
            logger.error("上传阿里云OSS服务器异常." + e.getMessage(), e);
        }

        return resultStr;
    }

    private static String dealFolder(String folder) {
        return StringUtils.trimFirstAndLastChar(folder, "/") + "/";
    }

    /**
     * OSS地址替换为cdn地址
     *
     * @param url        OSS返回url
     * @param bucketName
     * @param cdnUrl
     * @return
     */
    private static String dealResultUrl(String url, String bucketName, String cdnUrl) {
        //处理参数
        int idx = url.indexOf("?");
        if (idx > 0) {
            url = url.substring(0, idx);
        }
        if (StringUtils.isEmpty(cdnUrl)) {
            return url;
        }
        //替换为cdn地址
        return url.replace(bucketName + "." + OSSProperty.endpoint, cdnUrl);
    }

    public static String generateRandomFilename() {
        Random rand = new Random();//生成随机数
        int random = rand.nextInt();

        Calendar calCurrent = Calendar.getInstance();
        int intDay = calCurrent.get(Calendar.DATE);
        int intMonth = calCurrent.get(Calendar.MONTH) + 1;
        int intYear = calCurrent.get(Calendar.YEAR);
        return intYear + "_" + intMonth + "_" + intDay + "_" + (random > 0 ? random : (-1) * random);
    }


}
